题解 UVA12125 【March of the Penguins】

发表于 | 更新于

$Description$

给定一些冰块，每个冰块上有一些企鹅，每个冰块有一个可以从当前冰块跳出的次数限制，每个冰块位于一个坐标，现在每个企鹅跳跃力为$d$，问所有企鹅能否跳到一点上，如果可以输出所有落脚冰块，如果没有方案就打印$-1$

$Solution$

对于每个点$i$进行拆点,分成$i_1~$和$i_2~$，中间连一条容量为能跳出次数的边，从源点$s$连向$i_1~$，流量为每个点的企鹅数，然后$n^2$枚举点，如果两个点$i,j$距离$\leqslant d$，那么分别从$i_1~$连向$j_2~,j_1~$连向$i_2~$,容量为$inf$，接着枚举汇点$t$(因为每个点都可能成为企鹅的集合点)，将$t_1$作为汇点,接着跑最大流，看看是否满流，如果满流了就记录答案，最后判断一下是否有答案，没有答案输出$-1$即可

$Code$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#include <bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define pb push_back
#define N 40000
using namespace std;
struct edge{
    int to,dis,w,next;
}e[1007000];
inline int read(){
    int x=0,w=0;char ch=getchar();
    while (!isdigit(ch))w|=ch=='-',ch=getchar();
    while (isdigit(ch))x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return w?-x:x;
}
int cnt=1,head[N],dep[N],n,inque[N],vis[N],x[N],y[N],a[N],b[N],cur[N],cost,w[N],s,t;
inline void add(int u,int v,int d){
    e[++cnt].to=v;
    e[cnt].dis=d;
    e[cnt].next=head[u];
    head[u]=cnt;
    e[++cnt].to=u;
    e[cnt].dis=0;
    e[cnt].next=head[v];
    head[v]=cnt;
}
bool bfs(){
    memset(dep,0,sizeof(dep));
    queue<int>q;q.push(s);
    dep[s]=1;
    while (!q.empty()){
        int u=q.front();q.pop();inque[u]=0;
        for (int i=head[u];i;i=e[i].next){
            int v=e[i].to;
            if (e[i].dis&&!dep[v]){
                dep[v]=dep[u]+1;
                if (!inque[v])q.push(v),inque[v]=1;
            }
        }
    }
    return dep[t];
}
int dfs(int u,int mn){
    if (u==t)return mn;
    int used=0,mi;
    for (int &i=cur[u];i;i=e[i].next){
        int v=e[i].to;
        if (dep[v]==dep[u]+1&&e[i].dis)
            if (mi=dfs(v,min(e[i].dis,mn-used))){
                e[i].dis-=mi;
                e[i^1].dis+=mi;
                used+=mi;
                if (used==mn)break;
            }
    }
    return used;
}
int Dinic(){
    int res=0;
    while (bfs()){
        for (int i=0;i<=2*n;++i)cur[i]=head[i];
        res+=dfs(s,inf);
    }
    return res;
}
double d;
inline bool check(int a,int b){
    double res=sqrt(1.0*((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b])));
    if (res-d<1e-6)return 1;
    return 0;
}
void link(){
    memset(head,0,sizeof(head));cnt=1;
    for (int i=1;i<=n;++i)
        add(s,i,a[i]),add(i,i+n,b[i]);
    for (int i=1;i<=n;++i)
        for (int j=1;j<=n;++j)
            if (i!=j&&check(i,j))
                add(j+n,i,inf);
}
signed main(){
    int T=read();
    while (T--){
        memset(head,0,sizeof(head));cnt=1;int tot=0,mx=0;
        n=read();scanf("%lf",&d);
        s=0;
        for (int i=1;i<=n;++i)
            x[i]=read(),y[i]=read(),a[i]=read(),b[i]=read(),mx+=a[i];
        for (int i=1;i<=n;++i){
            t=i;
            link();
            if (Dinic()==mx){
                ++tot;
                if (tot==1)printf("%d",i-1);
                else printf(" %d",i-1);
            }
        }
        if (!tot)puts("-1");
        else puts("");
    }
    return 0;
}